package top.lshaci.learning.netty.chat.server;

import cn.hutool.core.date.DateTime;
import cn.hutool.core.util.StrUtil;
import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;
import lombok.extern.slf4j.Slf4j;

/**
 * GroupChatServerHander
 *
 * @author lshaci
 * @since 1.0.0
 */
@Slf4j
public class GroupChatServerHandler extends SimpleChannelInboundHandler<String> {

    /**
     * GlobalEventExecutor.INSTANCE 是全局的事件执行器，是一个单例
     */
    private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    @Override
    public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
        System.err.println("channelRegistered");
    }

    /**
     * handlerAdded 表示连接建立，一旦连接，第一个被执行
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        // 将该客户端加入聊天的信息推送给其它在线的客户端
        String msg = StrUtil.format("[客户端] {}: 加入聊天 {}\n", channel.remoteAddress(), DateTime.now());
        // 该方法会将 channelGroup 中所有的channel遍历，并发送消息
        // 不需要我们自己来进行遍历
        channelGroup.writeAndFlush(msg);
        channelGroup.add(channel);
    }

    /**
     * 连接端口，将客户端离开信息推送给当前在线的客户
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        String msg = StrUtil.format("[客户端] {}: 离开了 {}\n", channel.remoteAddress(), DateTime.now());
        channelGroup.writeAndFlush(msg);
        log.debug("Channel Group Size: {}", channelGroup.size());
    }

    /**
     * 表示 channel 处于活动状态，提示 客户端 上线
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        log.debug("{} 上线了.", ctx.channel().remoteAddress());
    }

    /**
     * 表示 channel 处于不活动状态，提示 客户端 离线了
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        log.debug("{} 离线了~", ctx.channel().remoteAddress());
    }

    /**
     * 读取数据
     *
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.forEach(c -> {
            String message;
            if (c == channel) {
                // 回显自己发送的消息
                message = StrUtil.format("[自己] {} : {}\n", channel.remoteAddress(), msg);
            } else {
                // 转发消息
                message = StrUtil.format("[客户] {} : {}\n", channel.remoteAddress(), msg);
            }
            c.writeAndFlush(message);
        });
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        cause.printStackTrace();
        ctx.close();
    }
}
